Skip to content

feat: add the implementation of Gemm operator on Cambricon#7

Merged
voltjia merged 6 commits intofeat/dev-infrafrom
feat/dev-gemm-cambricon
Mar 5, 2026
Merged

feat: add the implementation of Gemm operator on Cambricon#7
voltjia merged 6 commits intofeat/dev-infrafrom
feat/dev-gemm-cambricon

Conversation

@bitzyz
Copy link

@bitzyz bitzyz commented Mar 2, 2026

  1. Add Cambricon CMake options
  2. Implement Cambricon GEMM operator in src/cambricon/gemm/cnblas.h
  3. Add MLU skip logic in tests/test_gemm.py (transpose operations not currently supported)
  4. Add Cambricon test in examples/gemm/gemm.cc
  5. Add protected member variable workspace_size_ to Operator class
  6. Add Cmabricon common utils_func in src/cambricon/common.h

@bitzyz bitzyz self-assigned this Mar 2, 2026
@bitzyz bitzyz force-pushed the feat/dev-gemm-cambricon branch from 07207ac to a39d532 Compare March 2, 2026 07:09
@bitzyz
Copy link
Author

bitzyz commented Mar 2, 2026

image

@bitzyz bitzyz requested a review from voltjia March 2, 2026 07:10
@voltjia voltjia force-pushed the feat/dev-gemm-cambricon branch 2 times, most recently from 421b11e to 9ce1591 Compare March 5, 2026 12:01
@voltjia voltjia force-pushed the feat/dev-gemm-cambricon branch from 9ce1591 to e0fe60f Compare March 5, 2026 12:08
@voltjia
Copy link
Collaborator

voltjia commented Mar 5, 2026

image

@voltjia
Copy link
Collaborator

voltjia commented Mar 5, 2026

image

@voltjia voltjia merged commit 24cc11a into feat/dev-infra Mar 5, 2026
@voltjia voltjia deleted the feat/dev-gemm-cambricon branch March 5, 2026 12:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants